Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Don't check for donors when fitting MissingDataImputer (use the strategy to check) #338

Merged
merged 1 commit into from
Jun 26, 2024

Conversation

27pchrisl
Copy link
Contributor

Hi,

During the fit stage of MissingDataImputer, it checks if there are donors: "Dataset must contain at least 1 donor per feature column.".

This is true for every strategy except the Constant strategy, where you don't need donors.
I checked, and every other strategy does its own check, throwing the right Exception if the provided array is empty.

This PR removes the check within MissingDataImputer so it's possible to use the Constant strategy with a dataset that has no donors.

@andrewdalpino
Copy link
Member

Looks good to me, thank you for the great work @27pchrisl!

@andrewdalpino andrewdalpino merged commit 3b7032a into RubixML:master Jun 26, 2024
1 of 13 checks passed
@github-actions github-actions bot locked and limited conversation to collaborators Jun 26, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants